#include "Until.hpp"
#include <vector>

#ifndef __BUFFER_HPP__
#define __BUFFER_HPP__

#include <vector>
#include <algorithm>

namespace Logs
{
    #define DEFAULT_BUFFER_SIZE (10 * 1024 * 1024)
    // 扩容设置一个阈值，到达阈值之前可以二倍扩容。
    // 到达后就线性扩容。
    #define THRESHOLD_BUFFER_SIZE (80 * 1024 * 1024)
    #define INCREATE_BUFFER_SIZE (10 * 1024 * 1024)
    class Buffer
    {
    public:
        Buffer()
            :_buffer(DEFAULT_BUFFER_SIZE)
            ,_write_idx(0)
            ,_reader_idx(0)
        {}
        // 可读数据的起始位置。
        const char*  begin()
        {
            return &_buffer[_reader_idx];
        }
        // 可以读多少数据。
        size_t readAbleLen()
        {
            return _write_idx - _reader_idx;
        }
        // 还可以写入多少数据。
        size_t writeAbleLen()
        {
            return (_buffer.size() - _write_idx);
        }
        // 想缓冲区写数据。
        void push(const char* str,size_t len)
        {
            // 不扩容的。
            // if((len + _write_idx) >= writeAbleLen())
            // {
            //     return ;
            // }
            // 扩容
            if((len + _write_idx) >= writeAbleLen())
            {
                enhanceBuffer((len + _write_idx));
            }
            // 取出地址。
            std::copy(str,str + len,&_buffer[_write_idx]);
            // for(int i = 0;i < len;++i)
            // {
            //     _buffer[_write_idx] = str[i];
            //     moveWrite(1);
            // }
            moveWrite(len);
        }
        // 将指针移到后面
        void moveRead(size_t len)
        {
            _reader_idx += len;
        }
        // 重置读写指针，初始化缓冲区。
        void reset()
        {
            _reader_idx = 0;
            _write_idx = 0;
        }
        // 对Buffer做交换。
        void swap(Buffer& buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx,buffer._reader_idx);
            std::swap(_write_idx,buffer._write_idx);
        }
        // 判断缓冲区是否为空。
        bool empty()
        {
            return _write_idx == _reader_idx;
        }
    private:
        void enhanceBuffer(size_t len)
        {
            // 日志不可能会一下子就大到那种地步，没人会干。
            // 到达指定容量后扩容。
            if (len <= writeAbleLen()) return;
            /*每次增大1M大小*/
            size_t new_capacity;
            if (_buffer.size() < THRESHOLD_BUFFER_SIZE)
                new_capacity = _buffer.size() * 2 + len;
            else
                new_capacity = _buffer.size() + INCREATE_BUFFER_SIZE + len;
            _buffer.resize(new_capacity);
            // if(len < THRESHOLD_BUFFER_SIZE)
            // {
            //     _buffer.reserve(INCREATE_BUFFER_SIZE * 2);
            // }
            // else if((len + _write_idx) < DEFAULT_BUFFER_SIZE)
            // {
            //     _buffer.reserve(2 * _buffer.size());
            // }
            // else
            // {
            //     _buffer.reserve(_buffer.size());
            // }
        }
        void moveWrite(size_t len)
        {
            _write_idx += len;
        }
    private:
        std::vector<char> _buffer;
        size_t _reader_idx;// 可读取数据的指针 —————— 下标
        size_t _write_idx;// 可写入数据的指针。
    };
}

#endif